from network import QKDNetwork

n = QKDNetwork("generate.txt")
# n.showNetwork()
# print(n.getGreedyPathWithFlowMatrix(3, 4, [
#         [34, 54, 57, 55, 47],
#         [2, 4, 57, 8, 6],
#         [23, 4, 32, 40, 30],
#         [9, 12, 34, 5, 6],
#         [3, 4, 54, 6, 7],
#     ]))
print(n.getShortedPathWithOneBrokenLinkAndFlowMatrix(3, 4, [
        [34, 54, 57, 55, 47],
        [2, 4, 57, 8, 6],
        [1000, 1000, 1000, 4, 1000],
        [9, 12, 34, 5, 6],
        [3, 4, 54, 6, 7],
    ]))